Fix on DAG1 code