Fix minor issue on line 254