Fixed test case partially. Will run when Map node is migrated