Display mode switch fixes