fix(tests) - get test suite working for unit tests