Switch from assert to Chai expect