Challenge: Fix the Code

Here is a challenge to fix another bug in the code given in the previous lesson