diff --git a/autogpts/autogpt/tests/integration/test_image_gen.py b/autogpts/autogpt/tests/integration/test_image_gen.py
index c14d1c7c..28dedb9f 100644
--- a/autogpts/autogpt/tests/integration/test_image_gen.py
+++ b/autogpts/autogpt/tests/integration/test_image_gen.py
@@ -143,6 +143,7 @@ def test_huggingface_fail_request_with_delay(
             mock_post.return_value.text = return_text
 
         agent.legacy_config.image_provider = "huggingface"
+        agent.legacy_config.huggingface_api_token = "mock-api-key"
         agent.legacy_config.huggingface_image_model = image_model
         prompt = "astronaut riding a horse"