Switch to expect_msgs in tests.