Unit tests work fine FINALLY!