Updated tests to match