minor fix for migration script