Migration bug fix

Signed-off-by: Sheik Hassan <solergiga@yahoo.com>