adding assert for test