Fix linq issue