diff --git a/test_Fuji.py b/test_Fuji.py
index a713b01..79122ce 100644
--- a/test_Fuji.py
+++ b/test_Fuji.py
@@ -1,7 +1,7 @@
 from __future__ import division
 import os, scipy.io
 import tensorflow as tf
-import tensorflow.contrib.slim as slim
+import tf_slim as slim
 import numpy as np
 import rawpy
 import glob