[FLINK-1208] Fix test case