Udpating diag, to not throw an error