Revert "Remove unnecessary expect"

This reverts commit fda0b0d5ab6f0b20b3008bcc682e9921fd7f07fe.