Fix bug in assertCalled