Migration Lib exception fix

Migration Lib exception fix