fix this using relation.