Re-add missing save Ref code & msg fix