Switched this to use `calledWith`