Fixing display selector bug