fix compile error in diego test