bug fix

line 18