A few small fixes

I hope this is cool! :dancers: