diff --git a/src/ui/components/cards.py b/src/ui/components/cards.py
index 07e3ff8..de7da50 100644
--- a/src/ui/components/cards.py
+++ b/src/ui/components/cards.py
@@ -1,7 +1,7 @@
 """Card components for DevDash."""
 from textual.widgets import Static
 
-from src.models import IssueModel, PullRequestModel, WorkflowModel
+from src.models import IssueModel, PullRequestModel, WorkflowRunModel
 
 
 class PullRequestCard(Static):
@@ -115,7 +115,7 @@ class WorkflowCard(Static):
     }
     """
 
-    def __init__(self, workflow: WorkflowModel):
+    def __init__(self, workflow: WorkflowRunModel):
         """Initialize workflow card.
 
         Args: