Fix diff not returning issue