fix 1 minor bug