fix the bug of not work