fix building of diag

Change-Id: Id34c3f1bc50a0143ccd22c278c3d4901f73ad302