add back step in test