fixed bug on line 16