diff --git a/gso/cli/imports.py b/gso/cli/imports.py
index 9bdf8e79db93fc070c2125a6bb3bbd58a153b917..6aa55c5dfc6bb5701d4cf81f5db6d6a1c3244d12 100644
--- a/gso/cli/imports.py
+++ b/gso/cli/imports.py
@@ -6,7 +6,7 @@ from pathlib import Path
 from typing import TypeVar
 
 import typer
-import yaml  # type: ignore[import-untyped]
+import yaml
 from pydantic import ValidationError
 
 from gso.api.v1.imports import (
diff --git a/requirements.txt b/requirements.txt
index c2e839b44147878595c0f59457a07a06d7c90936..ca077d368e511aeb84a585f696abd685697eedcf 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,6 +9,7 @@ celery==5.3.4
 # Test and linting dependencies
 celery-stubs==0.1.3
 types-requests==2.31.0.1
+types-PyYAML==6.0.12.12
 pytest==7.4.3
 faker==19.13.0
 responses==0.24.0