Updated tests to use chai's expect module.