refactor(test): making stubbing easier