moar bug fixes