Added tests for multiple types returned from relation query