Removed unnecessary shouldBeCalled() calls