Fix record streaming test

This still looks *awful*, clean it up later