diff --git a/pipelines/data_integration.py b/pipelines/data_integration.py
index 5b9b529..d76f532 100644
--- a/pipelines/data_integration.py
+++ b/pipelines/data_integration.py
@@ -222,7 +222,7 @@ if __name__ == "__main__":
     # Handle function execution based on the provided function name
     if function_name == 'run':
 
-        if len(sys.argv) != 2:
+        if len(sys.argv) != 3:
             print("Usage: python data_integration.py run <path_to_config_yamlFile>")
             sys.exit(1)
         # Extract path to configuration file, specifying the data integration task