Fix bugs 2