Made the test step more correct