Step 1: Red - fix test