Updated tests
small change