1. fix minor bug