Fixes, cheers @ninjalulz