Fixes, cheers @EPuncker