Fixed Migration Bug