Update test to use JavaTestKit