Add wait in top stories test