Unit tests actually work now