Fixing error in line 5