diff --git a/lib/streaming_providers/base/models/streaming_channel.py b/lib/streaming_providers/base/models/streaming_channel.py
index 2edb743..ca54cf2 100644
--- a/lib/streaming_providers/base/models/streaming_channel.py
+++ b/lib/streaming_providers/base/models/streaming_channel.py
@@ -4,7 +4,7 @@ from dataclasses import dataclass
 from enum import Enum
 from typing import Dict, List, Optional
 
-from .drm_models import DRMConfig
+from .drm import DRMConfig
 
 # Setup logging for validation warnings
 logger = logging.getLogger(__name__)