Correcting the bug