Add the missing statement back