Bugs Fixes

Bug Fixes