Use migration instance methods

Resolves #444