Driving automation introduces multiple driving modes to maximize the system’s benefits, but drivers must monitor and stay aware of these modes, which can sometimes lead to mode confusion. We modified Degani and Heymann’s state diagram method to assess the mode structure of a hypothetical driving automation system and the likelihood of discrepancies between drivers’ mode awareness and the system’s actual mode. We used the modified method and driving simulation data from participants who weren’t fully informed about all automation modes. The modified method visualized all possible combinations of automation modes and drivers’ mode awareness, highlighting where they diverge and estimating the frequency of the divergence. The diagram identified areas where human-machine interface (HMI) design can help drivers maintain accurate mode awareness. The modified state diagram method provides actionable insights for designing HMIs to reduce mode confusion and can be used to develop computational models of automation mode structures.