Adapt to latest HaxePunk