Removed a wrong import

There was still a breaking test.