Fix logic error to work with arg-taking modes.