add assert to test