Change diameterstack to allow mocking it out