Tiny migration fix