JOJ-87 Unit test fixes