Only wait for the finished message in the updown test

Previously any unexpected messages would crash, but I think just
blocking until the finish message comes in is enough.