Corrected the bug 2