fix unit tests for stubbing