Small bug fix in diff