Adding in assertion