Updated for relations api