Step 4: Red - fix test