Fix import after move