fix for migration script