diff --git a/.github/workflows/.github-ci.yml b/.github/workflows/.github-ci.yml
index 79c6a0c..fe31ed0 100644
--- a/.github/workflows/.github-ci.yml
+++ b/.github/workflows/.github-ci.yml
@@ -45,6 +45,7 @@ jobs:
         run: |
           python3 -m pip install --upgrade pip
           pip3 install -r requirements.txt
+          pip3 install google-cloud-aiplatform
       - name: Run unit tests
         run: |
           pytest tests/unit --cov=AutoMLOps=
diff --git a/tests/integration/test_beans_training_model.py b/tests/integration/test_beans_training_model.py
index f2df8e7..57add4a 100644
--- a/tests/integration/test_beans_training_model.py
+++ b/tests/integration/test_beans_training_model.py
@@ -181,8 +181,6 @@ def pipeline(bq_table: str,
 
 
 def test_beans_training_model():
-    # Importing here to avoid conflict with repeated import in deploy_model
-
     project_id = os.environ.get('PROJECT_ID')
     model_id = 'dry-beans-dt'
     ## Define the Pipeline Arguments