Minor fix to balloon framework test.