Fix unhandled keep_running argument issue