diff --git a/openpype/hosts/tvpaint/api/pipeline.py b/openpype/hosts/tvpaint/api/pipeline.py
index 6a729e39c3..3794bf2e24 100644
--- a/openpype/hosts/tvpaint/api/pipeline.py
+++ b/openpype/hosts/tvpaint/api/pipeline.py
@@ -124,8 +124,11 @@ class TVPaintHost(HostBase, IWorkfileHost, ILoadHost, IPublishHost):
         if not context:
             return get_global_context()
 
+        if "project_name" in context:
+            return context
+        # This is legacy way how context was stored
         return {
-            "project_name": context["project"],
+            "project_name": context.get("project"),
             "asset_name": context.get("asset"),
             "task_name": context.get("task")
         }