Updated tests .expect