fix minor bug again