fix small bug for code style