fixing the bug？