Fix Flume unit tests