Cleaning up test runner.

Looks like the JVM is running its completion callback multiple times. This shouldn't be possible... time to investigate!