Adding assertion flow