Fix: don't save last layer twice.