Small refactor - Fixing an unsaved objects module state