fix a bug

not test yet.