fixed on minor bug