fix minor bug on line 157