Fixed a little bug in the belongs to