Moar bugfixes