solve minor bug