Fix complexity complain