Fix for 'I should see' step

This fixes the 'I should see' step... and doesn't seem to break anything else.