Forgot to remove import of mock from client