fixed a bug 2