fixing bug
Fixed the bug