Fix broken test made after changing the dialogue screen's states-per-row count from 4 to 3