fix overlooked changes