Forgot to change import after rename