Add diag message for test