Fixed a test bug in remove base step.