Post review improvement - missed one line