Fix bugs Part2