Bugs fixes 2