Some mini fixes I suppose