Updated before_migrate to link assets