diff --git a/services/backend/tasks.py b/services/backend/tasks.py
index b634d35..ef94d90 100644
--- a/services/backend/tasks.py
+++ b/services/backend/tasks.py
@@ -9,7 +9,7 @@ import os
 from invoke import task
 from invoke.exceptions import UnexpectedExit
 
-from hotpocket_workspace_tools import get_workspace_mode
+from hotpocket_workspace_tools import get_workspace_mode, WorkspaceMode
 
 WORKSPACE_MODE = get_workspace_mode()
 ENV = os.getenv('HOTPOCKET_BACKEND_ENV', 'docker')
@@ -110,7 +110,7 @@ def setup(ctx):
     ctx.run('python manage.py migrate')
     ctx.run('python manage.py create_initial_account hotpocket hotpocketm4st3r')
 
-    if WORKSPACE_MODE == WORKSPACE_MODE.METAL:
+    if WORKSPACE_MODE == WorkspaceMode.METAL:
         ctx.run('mkdir -p run/uploads')