Forgot to change an import.
(strange, all tests passed even with that incorrect import...)