Migration bugfix