can we try to use .was.called() here?