Fixes migration dependency