fix a bug :/