Tweak migration check