diff --git a/src/mflux/config/runtime_config.py b/src/mflux/config/runtime_config.py
index a4f6ecd..018f0c5 100644
--- a/src/mflux/config/runtime_config.py
+++ b/src/mflux/config/runtime_config.py
@@ -1,5 +1,4 @@
 import logging
-import typing as t
 
 import mlx.core as mx
 import numpy as np
@@ -9,8 +8,6 @@ from mflux.config.model_config import ModelConfig
 
 logger = logging.getLogger(__name__)
 
-VAE: t.TypeAlias = "mflux.models.vae.vae.VAE"  # noqa: F821
-
 
 class RuntimeConfig:
     def __init__(self, config: Config | ConfigControlnet, model_config):