fixed a typo in the TODO
added a nice convenience function for setting tests