Fix bug on line 14