added migration and solved load issue