minor fix to latest migration

bodged it on first try. here the fix.