Test should reflect actual state structure, right?