work on the bug