Bug fixed

- Unit test now works