Fixed the bug mentioned earlier