Fix issue with finding migrations in the proper assembly