diff --git a/docs/courses/mlops/jobs-and-services.md b/docs/courses/mlops/jobs-and-services.md
index b64459d9..9ca565d1 100644
--- a/docs/courses/mlops/jobs-and-services.md
+++ b/docs/courses/mlops/jobs-and-services.md
@@ -157,6 +157,11 @@ Similar to Anyscale Jobs, we'll start by creating a [`serve_model.py`](https://g
 ```python linenums="1"
 # deploy/services/serve_model.py
 
+import os
+import subprocess
+from madewithml.config import MODEL_REGISTRY  # NOQA: E402
+from madewithml.serve import ModelDeployment  # NOQA: E402
+
 # Copy from S3
 github_username = os.environ.get("GITHUB_USERNAME")
 subprocess.check_output(["aws", "s3", "cp", f"s3://madewithml/{github_username}/mlflow/", str(MODEL_REGISTRY), "--recursive"])