Updated tests

^