Code wrapper; frustrated by unit testing setup