Add correct migration