diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 0a81ca4..3dbe72b 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -35,3 +35,4 @@ jobs:
         with:
           stack-name: ${{ matrix.stack-name }}
           python-version: ${{ matrix.python-version }}
+          ref-zenml: feature/OSS-2300-model-watch-tower-v0.1
diff --git a/template/pipelines/batch_inference.py b/template/pipelines/batch_inference.py
index b19c3a7..8f9c26c 100644
--- a/template/pipelines/batch_inference.py
+++ b/template/pipelines/batch_inference.py
@@ -20,7 +20,7 @@
     mlflow_model_registry_deployer_step,
 )
 from zenml.logger import get_logger
-from zenml.steps.external_artifact import ExternalArtifact
+from zenml.artifacts.external_artifact import ExternalArtifact
 
 logger = get_logger(__name__)
 
diff --git a/template/pipelines/training.py b/template/pipelines/training.py
index 8939914..d80083d 100644
--- a/template/pipelines/training.py
+++ b/template/pipelines/training.py
@@ -31,7 +31,7 @@
 )
 from zenml.integrations.mlflow.steps.mlflow_registry import mlflow_register_model_step
 from zenml.logger import get_logger
-from zenml.steps.external_artifact import ExternalArtifact
+from zenml.artifacts.external_artifact import ExternalArtifact
 
 
 logger = get_logger(__name__)
diff --git a/template/run.py b/template/run.py
index 562fb91..49fc089 100644
--- a/template/run.py
+++ b/template/run.py
@@ -1,7 +1,7 @@
 # {% include 'template/license_header' %}
 
 
-from zenml.steps.external_artifact import ExternalArtifact
+from zenml.artifacts.external_artifact import ExternalArtifact
 from zenml.logger import get_logger
 from pipelines import {{product_name}}_batch_inference, {{product_name}}_training
 from config import MetaConfig
@@ -177,7 +177,7 @@ def main(
     logger.info(
         "Batch inference pipeline finished successfully! "
         "You can find predictions in Artifact Store using ID: "
-        f"`{str(artifact.upload_if_necessary())}`."
+        f"`{str(artifact.id)}`."
     )