Fixed stale method after run_command changes.