Fix review issue