Fix test to support Steps