Fix for diff issue