Fix unit test build.
Merge pull request #764 from brendandburns/expect

Fix unit test build.