ForeignKey fix