diff --git a/tests/codegen/sample_workflows.py b/tests/codegen/sample_workflows.py
index 17a024e..f39d8c5 100644
--- a/tests/codegen/sample_workflows.py
+++ b/tests/codegen/sample_workflows.py
@@ -5,6 +5,7 @@
 from brickflow import JarTaskLibrary, PypiTaskLibrary
 from brickflow.bundles.model import (
     JobsContinuous,
+    JobsParameters,
     JobsTasksRunJobTaskPipelineParams,
     JobsTasksSqlTaskAlert,
     JobsTasksSqlTaskDashboard,