fixed as per comment in pr