Small bug fix for relationship extension