Fixing bug in the code