Migration bugfix

Signed-off-by: Nick <njkt809q@gmail.com>