step 6: Red - fixed test