diff --git a/python-malformed-yaml/Dockerfile b/python-malformed-yaml/Dockerfile
index b76713d9..9583622f 100644
--- a/python-malformed-yaml/Dockerfile
+++ b/python-malformed-yaml/Dockerfile
@@ -1,8 +1,6 @@
 FROM python:3.12.0-alpine3.17
 
-# We are installing a dependency here directly into our app source dir
-RUN pip install pathlib
-RUN pip install yaml
+RUN pip install PyYaml
 
 COPY main.py /main.py
 
diff --git a/python-malformed-yaml/main.py b/python-malformed-yaml/main.py
index c85505b9..412f3e08 100644
--- a/python-malformed-yaml/main.py
+++ b/python-malformed-yaml/main.py
@@ -1,5 +1,4 @@
 import yaml
-import pathlib
 
 
 def main():