Add missing expectation

https://trello.com/c/emi3eU8U