fix: ensure migrations are context aware

fix: ensure migrations are context aware #256