diff --git a/duplicate.py b/duplicate.py
index e7ac5767d8b2c..ddbd46b8e8d61 100644
--- a/duplicate.py
+++ b/duplicate.py
@@ -1,6 +1,7 @@
 import importlib
 import os
 import glob
+import sys
 
 
 def get_all_functions_from_directory(root_dir, startswith="test"):
@@ -40,4 +41,4 @@ def check_duplicate():
     common_set = check_duplicate()
     if len(common_set) != 0:
         print("This function already exists in the functional API.")
-        exit(1)
+        sys.exit(1)