Corrected relationship retrieval error

Integration tests failing. Highlighted issue. Keep integration tests up
to date as this was invaluable