Return the correct address family
Add tests