diff --git a/app/components/chat/Artifact.tsx b/app/components/chat/Artifact.tsx
index 75cf6e26..537a6a63 100644
--- a/app/components/chat/Artifact.tsx
+++ b/app/components/chat/Artifact.tsx
@@ -111,7 +111,7 @@ export const Artifact = memo(({ messageId }: ArtifactProps) => {
             </div>
             <div className="text-bolt-elements-textPrimary font-medium leading-5 text-sm">
               {allActionFinished
-                ? artifact.id === 'imported-files'
+                ? artifact.id === 'restored-project-setup'
                   ? 'Restore files from snapshot'
                   : 'Initial files created'
                 : 'Creating initial files'}