Tiny style fix and added a TODO.