Try to fix migrations

Signed-off-by: Adam Lord <adam.lord@gmail.com>