Fixes #183 -- Add migration dependency