Step 6: Red - fixed test