U_matrix fix