diff --git a/resnet.py b/resnet.py
index a056c41..24801dc 100644
--- a/resnet.py
+++ b/resnet.py
@@ -101,7 +101,10 @@ def inference_small(x,
     c['fc_units_out'] = num_classes
     c['num_blocks'] = num_blocks
     c['num_classes'] = num_classes
-    inference_small_config(x, c)
+    x = inference_small_config(x, c)
+    
+    return x
+    
 
 def inference_small_config(x, c):
     c['bottleneck'] = False