Fixing bug

Fixing bug