Updated tests assertions to work correctly