Fixes migration gen bug