Fix input tests to work with master gemini