Fixes

Moar fixes