Fix migrations (/cc @mekanics)