fix a bug

by wujun