Add setter for test injection