bugs fixes

bugs fixes