Remove max_runtime argument