fix last minor bug