fix the bug #2