oops, fix overlooked TODO