Unit tests now work!