Unit Tests

- Corrected unit test, which had the wrong expected output.