Fix bugs 4