diff --git a/src/primaite/game/agent/interface.py b/src/primaite/game/agent/interface.py
index 38116987..b321b17c 100644
--- a/src/primaite/game/agent/interface.py
+++ b/src/primaite/game/agent/interface.py
@@ -1,9 +1,8 @@
 """Interface for agents."""
 import random
 from abc import ABC, abstractmethod
-from typing import Dict, List, Optional, Tuple, TYPE_CHECKING, TypeAlias, Union
+from typing import Dict, List, Optional, Tuple, TYPE_CHECKING
 
-import numpy as np
 from gymnasium.core import ActType, ObsType
 from pydantic import BaseModel
 
@@ -14,8 +13,6 @@ from primaite.game.agent.rewards import RewardFunction
 if TYPE_CHECKING:
     from primaite.simulator.system.services.red_services.data_manipulation_bot import DataManipulationBot
 
-ObsType: TypeAlias = Union[Dict, np.ndarray]
-
 
 class AgentStartSettings(BaseModel):
     """Configuration values for when an agent starts performing actions."""
diff --git a/src/primaite/simulator/system/applications/database_client.py b/src/primaite/simulator/system/applications/database_client.py
index da2299c4..3c4f1b75 100644
--- a/src/primaite/simulator/system/applications/database_client.py
+++ b/src/primaite/simulator/system/applications/database_client.py
@@ -148,6 +148,7 @@ class DatabaseClient(Application):
             # Initialise the tracker of this ID to False
             self._query_success_tracker[query_id] = False
             return self._query(sql=sql, query_id=query_id)
+        return False
 
     def receive(self, payload: Any, session_id: str, **kwargs) -> bool:
         """