tests: Really fix import.