[BUGFIX] Fixed model saving