Add child classes to rebuild_fks function