diff --git a/atst/domain/csp/cloud/azure_cloud_provider.py b/atst/domain/csp/cloud/azure_cloud_provider.py
index ce6d45ee..79f352d5 100644
--- a/atst/domain/csp/cloud/azure_cloud_provider.py
+++ b/atst/domain/csp/cloud/azure_cloud_provider.py
@@ -27,7 +27,6 @@ from .models import (
     EnvironmentCSPPayload,
     EnvironmentCSPResult,
     KeyVaultCredentials,
-    ManagementGroupCSPResponse,
     ProductPurchaseCSPPayload,
     ProductPurchaseCSPResult,
     ProductPurchaseVerificationCSPPayload,
@@ -155,8 +154,6 @@ class AzureCloudProvider(CloudProviderInterface):
             payload.parent_id,
         )
 
-        if
-
         return EnvironmentCSPResult(**response)
 
     def create_atat_admin_user(
diff --git a/atst/domain/csp/cloud/mock_cloud_provider.py b/atst/domain/csp/cloud/mock_cloud_provider.py
index 96648b52..4955f327 100644
--- a/atst/domain/csp/cloud/mock_cloud_provider.py
+++ b/atst/domain/csp/cloud/mock_cloud_provider.py
@@ -4,9 +4,7 @@ from .cloud_provider_interface import CloudProviderInterface
 from .exceptions import (
     AuthenticationException,
     AuthorizationException,
-    BaselineProvisionException,
     ConnectionException,
-    EnvironmentCreationException,
     GeneralCSPException,
     UnknownServerException,
     UserProvisioningException,
diff --git a/tests/domain/cloud/test_azure_csp.py b/tests/domain/cloud/test_azure_csp.py
index 639997b3..2b55d1a5 100644
--- a/tests/domain/cloud/test_azure_csp.py
+++ b/tests/domain/cloud/test_azure_csp.py
@@ -61,7 +61,7 @@ def test_create_environment_succeeds(mock_azure: AzureCloudProvider):
     environment = EnvironmentFactory.create()
     mock_management_group_create(mock_azure, {"id": "Test Id"})
 
-    mock_azure = mock_get_secret(mock_azure, lambda *a, **k: json.dumps(MOCK_CREDS))
+    mock_azure = mock_get_secret(mock_azure)
 
     payload = EnvironmentCSPPayload(
         tenant_id="1234", display_name=environment.name, parent_id=str(uuid4())
diff --git a/tests/test_jobs.py b/tests/test_jobs.py
index a1a821e6..8e3d5b1d 100644
--- a/tests/test_jobs.py
+++ b/tests/test_jobs.py
@@ -155,7 +155,7 @@ def test_dispatch_create_environment(session, monkeypatch):
     # Given that I have a portfolio with an active CLIN and two environments,
     # one of which is deleted
     portfolio = PortfolioFactory.create(
-        applications=[{"environments": [{}, {}]}],
+        applications=[{"environments": [{}, {}], "cloud_id": uuid4().hex}],
         task_orders=[
             {
                 "create_clins": [