Change to only inject CountDownLatch object.