Migration code working