better logic to communicate to tests after the code has been injected