fix a bug more