diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py
index 1e3f5455a..e041ec0af 100644
--- a/litellm/tests/test_embedding.py
+++ b/litellm/tests/test_embedding.py
@@ -589,7 +589,7 @@ async def test_triton_embeddings():
         print(f"response: {response}")
 
         # stubbed endpoint is setup to return this
-        assert response.data[0]["embedding"] == [0.1, 0.2, 0.3]
+        assert response.data[0]["embedding"] == [0.1, 0.2]
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")