Entity framework refference fix