minor code improvement that was missed in the previous commit