Got steps working with expect instead of should