Make the selection highlight square again