Fixing Small Issue