Fix renaming files with custom model

Fix renaming files with custom model #324