Small Fix
Bug fixing