Add migration for sti