diff --git a/scripts/aws/ec2.py b/scripts/aws/ec2.py
index 9bb8d42c9..e5f9ebd19 100644
--- a/scripts/aws/ec2.py
+++ b/scripts/aws/ec2.py
@@ -16,7 +16,13 @@
 import yaml
 
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-from confidential_compute import ConfidentialCompute, AWSConfidentialComputeConfig, SecretNotFoundException
+from confidential_compute import ConfidentialCompute, ConfidentialComputeConfig, SecretNotFoundException
+
+
+class AWSConfidentialComputeConfig(ConfidentialComputeConfig):
+    enclave_memory_mb: int
+    enclave_cpu_count: int
+
 
 class EC2(ConfidentialCompute):
 
diff --git a/scripts/confidential_compute.py b/scripts/confidential_compute.py
index 325844513..395c04f49 100644
--- a/scripts/confidential_compute.py
+++ b/scripts/confidential_compute.py
@@ -12,10 +12,6 @@ class ConfidentialComputeConfig(TypedDict):
     core_base_url: str
     optout_base_url: str
     environment: str
-
-class AWSConfidentialComputeConfig(ConfidentialComputeConfig):
-    enclave_memory_mb: int
-    enclave_cpu_count: int
     
 class ConfidentialCompute(ABC):