Updated integration tests to use migration DSL