Fixed imports (dunno why the tests passed).