Fix bug with result class display