Add TODO note on tests