Fixed a bug

- The JUnit test now works.