Fixed tests

- Updated tests to reflect amended copy