Fixes migration issue