Cleanup use of migrations library