fix a bug 2