modify tests to include stubbing