diff --git a/du5/program.c b/du5/program.c
index 5d55eec..9005c8b 100644
--- a/du5/program.c
+++ b/du5/program.c
@@ -1,6 +1,7 @@
 #include<stdio.h>
 #include <string.h>
 #include <stdbool.h>
+#include <math.h>
 
 struct celovek{     ////definovanie štruktúry s názvom 'celovek'
     int golos;
@@ -49,7 +50,7 @@ void readDellSame(bool odinakovyi, bool dobre, int* number_celovek, struct celov
 void sort(int number_celovek, struct celovek celovek[50]){          //funkcia na triedenie poľa 'celovek' na základe počtu 'golov' a 'name'
     for (size_t i = 0; i <number_celovek; i++){
         for (size_t j = i; j < number_celovek; j++){
-            if (celovek[i].golos == celovek[j].golos){
+            if (isfinite(celovek[i].golos) == celovek[j].golos){
                 if(celovek[i].name[1] > celovek[j].name[1]){
                         int tmp = celovek[i].golos;
                         char tmp_name[60];