Update outdated gemini tests

/cc @dfilatov