diff --git a/.gitignore b/.gitignore
index fd64d3314e..79226a4920 100644
--- a/.gitignore
+++ b/.gitignore
@@ -90,4 +90,4 @@ ENV/
 
 # pyenv
 .python-version
-.vscode/
+.vscode/
\ No newline at end of file
diff --git a/tests_e2e/tests/scripts/agent_cpu_quota-check_agent_cpu_quota.py b/tests_e2e/tests/scripts/agent_cpu_quota-check_agent_cpu_quota.py
index 63871b43a6..c8aad49f59 100755
--- a/tests_e2e/tests/scripts/agent_cpu_quota-check_agent_cpu_quota.py
+++ b/tests_e2e/tests/scripts/agent_cpu_quota-check_agent_cpu_quota.py
@@ -169,7 +169,8 @@ def verify_process_check_on_agent_cgroups():
     #     [CGroupsException] The agent's cgroup includes unexpected processes: ['[PID: 25826] python3\x00/home/nam/Compute-Runtime-Tux-Pipeline/dungeon_crawler/s']
     wait_for_log_message(
         "Disabling resource usage monitoring. Reason: Check on cgroups failed:.+The agent's cgroup includes unexpected processes")
-    if not check_agent_quota_disabled():
+    disabled: bool = retry_if_false(check_agent_quota_disabled)
+    if not disabled:
         fail("The agent did not disable its CPUQuota: {0}".format(get_agent_cpu_quota()))
 
 
@@ -199,7 +200,8 @@ def verify_throttling_time_check_on_agent_cgroups():
         timeout=datetime.timedelta(minutes=10))
     wait_for_log_message("Stopped tracking cgroup walinuxagent.service", timeout=datetime.timedelta(minutes=10))
     wait_for_log_message("Executing systemctl daemon-reload...", timeout=datetime.timedelta(minutes=5))
-    if not check_agent_quota_disabled():
+    disabled: bool = retry_if_false(check_agent_quota_disabled)
+    if not disabled:
         fail("The agent did not disable its CPUQuota: {0}".format(get_agent_cpu_quota()))