Fix model drag and drop