Fix tests for gemini