Use ContextTypeAttribute during migration discovery

/cc @rowanmiller & @emilcicos