Fix top level unit info being wrong after publication view