add assert to tests