Fix big bugs