diff --git a/notebook.ipynb b/notebook.ipynb
index c315d79..d09da21 100644
--- a/notebook.ipynb
+++ b/notebook.ipynb
@@ -309,7 +309,7 @@
     "    skip_and_import_local_data(language)\n",
     "    display(Markdown(\"You choose to load the data from your local storage, you can skip the calls generation and move to [calls analysis](#calls_analysis)\"))\n",
     "else:\n",
-    "    create_tables()"
+    "    create_tables(project_name=project_name)"
    ]
   },
   {
diff --git a/project.yaml b/project.yaml
index aaada56..545db35 100644
--- a/project.yaml
+++ b/project.yaml
@@ -61,7 +61,7 @@ spec:
     source_code_target_dir: /home/mlrun_code/
     image: .mlrun-project-image-call-center-demo
     base_image: mlrun/mlrun-kfp
-  origin_url: git://github.com/mlrun/demo-call-center.git#refs/heads/main
+  origin_url: git://github.com/mlrun/demo-call-center.git#guy1992l-correction
   workflows:
   - image: .mlrun-project-image-call-center-demo
     name: calls-generation
@@ -71,10 +71,10 @@ spec:
     path: ./src/workflows/calls_analysis.py
   params:
     build_image: false
-    source: git://github.com/mlrun/demo-call-center.git#main
+    source: git://github.com/mlrun/demo-call-center.git#guy1992l-correction
     gpus: 0
     use_sqlite: false
     default_image: .mlrun-project-image-call-center-demo
-  source: git://github.com/mlrun/demo-call-center.git#main
+  source: git://github.com/mlrun/demo-call-center.git#guy1992l-correction
   conda: ''
   default_image: .mlrun-project-image-call-center-demo
diff --git a/project_setup.py b/project_setup.py
index 5ce1628..a3df31b 100644
--- a/project_setup.py
+++ b/project_setup.py
@@ -118,7 +118,7 @@ def setup(
     app.save()
 
     # Create the DB tables:
-    create_tables()
+    create_tables(project_name=project.name)
 
     # Save and return the project:
     project.save()
@@ -357,4 +357,4 @@ def _set_workflows(project: mlrun.projects.MlrunProject, image):
     )
     project.set_workflow(
         name="calls-analysis", workflow_path="./src/workflows/calls_analysis.py", image=image
-    )
\ No newline at end of file
+    )
diff --git a/src/calls_analysis/db_management.py b/src/calls_analysis/db_management.py
index 1aed185..cd4ab0e 100644
--- a/src/calls_analysis/db_management.py
+++ b/src/calls_analysis/db_management.py
@@ -219,12 +219,13 @@ def __del__(self):
                 pass
 
 
-def create_tables():
+def create_tables(project_name: str):
     """
     Create the call center schema tables for when creating or loading the MLRun project.
     """
     # Create an engine:
-    engine = DBEngine(mlrun.get_or_create_ctx("create_tables"))
+    engine = DBEngine(mlrun.get_or_create_ctx("create_tables", project=project_name))
+    
     # Create the schema's tables
     Base.metadata.create_all(engine.engine)