fixes...imports, missing-method after running tests