Adding missing assignment after last change to layout switching