Unit tests are now working