diff --git a/app/chain/mediaserver.py b/app/chain/mediaserver.py
index 650dbc9e7..ce33d35ad 100644
--- a/app/chain/mediaserver.py
+++ b/app/chain/mediaserver.py
@@ -1,6 +1,6 @@
 import threading
 from datetime import datetime
-from typing import Callable, Dict, List, Union, Optional, Generator, Any
+from typing import Callable, Dict, List, Union, Optional, Generator, Any, Tuple
 
 from app.chain import ChainBase
 from app.runtime.config import global_vars
diff --git a/app/chain/transfer.py b/app/chain/transfer.py
index a459f2b79..2b238ea46 100755
--- a/app/chain/transfer.py
+++ b/app/chain/transfer.py
@@ -14,7 +14,7 @@ from app.chain.media import MediaChain
 from app.chain.storage import StorageChain
 from app.chain.tmdb import TmdbChain
 from app.runtime.config import global_vars
-from app.domain.context import MediaInfo, MusicInfo
+from app.domain.context import MediaInfo, MusicInfo, TorrentInfo
 from app.runtime.events import eventmanager
 from app.domain.meta.metabase import MetaBase
 from app.domain.meta.metamusic import MetaMusic
diff --git a/tests/test_mediaserver_sync_incremental.py b/tests/test_mediaserver_sync_incremental.py
index 7eef264db..6054fd956 100644
--- a/tests/test_mediaserver_sync_incremental.py
+++ b/tests/test_mediaserver_sync_incremental.py
@@ -8,7 +8,7 @@ from sqlalchemy.orm import sessionmaker
 from app import schemas
 from app.chain import mediaserver as MEDIA_SERVER_CHAIN_MODULE
 from app.chain.mediaserver import MediaServerChain
-from app.db import Base
+from app.db.base import Base
 from app.db.oper.mediaserver import MediaServerOper
 from app.db.models.mediaserver import MediaServerItem
 from app.runtime.config import global_vars