Moar bug fixes