Add missing entrypoint arg