diff --git a/mmdnn/conversion/examples/caffe/extract_model.py b/mmdnn/conversion/examples/caffe/extract_model.py
index 8c7048b..db5b78e 100644
--- a/mmdnn/conversion/examples/caffe/extract_model.py
+++ b/mmdnn/conversion/examples/caffe/extract_model.py
@@ -64,7 +64,6 @@ def _main():
         # net = caffe.Net(arch_fn, weight_fn, caffe.TEST)
         func = TestKit.preprocess_func['caffe'][args.network]
         img = func(args.image)
-        print(img.size)
         img = np.transpose(img, (2, 0, 1))
         img = np.expand_dims(img, 0)
         net.blobs['data'].data[...] = img
diff --git a/tests/test_conversion_imagenet.py b/tests/test_conversion_imagenet.py
index 8553ee3..4a0eac4 100644
--- a/tests/test_conversion_imagenet.py
+++ b/tests/test_conversion_imagenet.py
@@ -275,15 +275,15 @@ class TestModels(CorrectnessTest):
             'imagenet1k-inception-bn'   : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
             'imagenet1k-resnet-152'     : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
             'squeezenet_v1.1'           : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
-            'imagenet1k-resnext-101-64x4d' : [TensorflowEmit, PytorchEmit], # TODO: CntkEmit
-            'imagenet1k-resnext-50'        : [TensorflowEmit, KerasEmit, PytorchEmit], # TODO: CntkEmit
+            'imagenet1k-resnext-101-64x4d' : [CntkEmit, TensorflowEmit, PytorchEmit], # Keras is too slow
+            'imagenet1k-resnext-50'        : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
         },
         'caffe' : {
-            # 'vgg19'         : [KerasEmit],
+            'vgg19'         : [KerasEmit],
             # 'alexnet'       : [KerasEmit],
-            # 'inception_v1'  : [CntkEmit],
-            # 'resnet152'     : [CntkEmit],
-            # 'squeezenet'    : [CntkEmit]
+            'inception_v1'  : [CntkEmit],
+            'resnet152'     : [CntkEmit],
+            'squeezenet'    : [CntkEmit]
         }
     }
 
@@ -339,7 +339,6 @@ class TestModels(CorrectnessTest):
 
                 self._compare_outputs(original_predict, converted_predict)
 
-            os.remove(self.tmpdir + network_name + "_converted.json")
             os.remove(self.tmpdir + network_name + "_converted.pb")
             os.remove(self.tmpdir + network_name + "_converted.npy")
             print("Testing {} model {} passed.".format(original_framework, network_name))
@@ -367,7 +366,7 @@ class TestModels(CorrectnessTest):
 
                 self._compare_outputs(original_predict, converted_predict)
 
-
+            os.remove(self.tmpdir + network_name + "_converted.json")
             os.remove(self.tmpdir + network_name + "_converted.pb")
             os.remove(self.tmpdir + network_name + "_converted.npy")
             print("Testing {} model {} passed.".format(original_framework, network_name))