Add flush test