fix the bug #5