Fix lint from previous PR