Fixed an import error.
Added tests for the existing relation work.