Fix DB migration