diff --git a/src/scgenerator/env.py b/src/scgenerator/env.py
index bcaca9a..e6bf2a6 100644
--- a/src/scgenerator/env.py
+++ b/src/scgenerator/env.py
@@ -2,6 +2,7 @@ from enum import Enum
 from pathlib import Path
 from typing import Any
 
+import matplotlib.pyplot as plt
 from pydantic import ImportString, NonNegativeFloat, ValidationInfo, model_validator
 from pydantic_settings import BaseSettings, SettingsConfigDict