Updated migration script
Fixes
Fixed organization edition