Fix compatibility issues with new HaxePunk