diff --git a/object_detection.py b/object_detection.py
index a549c92..2747c26 100644
--- a/object_detection.py
+++ b/object_detection.py
@@ -62,7 +62,7 @@ def download_model():
         tar_file = tarfile.open(model_file)
         for file in tar_file.getmembers():
           file_name = os.path.basename(file.name)
-          if 'toy_frozen_inference_graph.pb' in file_name:
+          if 'frozen_inference_graph.pb' in file_name:
             tar_file.extract(file, os.getcwd() + '/models/')
         os.remove(os.getcwd() + '/' + model_file)
     else:
diff --git a/object_detection_kcf_test.py b/object_detection_kcf_test.py
index 654dd20..70ab2b8 100644
--- a/object_detection_kcf_test.py
+++ b/object_detection_kcf_test.py
@@ -69,7 +69,7 @@ def download_model():
         tar_file = tarfile.open(model_file)
         for file in tar_file.getmembers():
           file_name = os.path.basename(file.name)
-          if 'toy_frozen_inference_graph.pb' in file_name:
+          if 'frozen_inference_graph.pb' in file_name:
             tar_file.extract(file, os.getcwd() + '/models/')
         os.remove(os.getcwd() + '/' + model_file)
     else: