Got rid of one minor TODO