Add a small migration step for testing