Fix test to support Steps
Merge pull request #17 from shmed/master

Fix test to support Steps