Fix migration error

Signed-off-by: Marco Rivadeneyra <rockiano@gmail.com>