fixing the bug?