Fixed bug in line 155