Adding a minor TODO