Modified the unit test to only assert that the mocks were called successfully (i.e. only one thing!)