diff --git a/Web/Models/Entities/TicketComment.php b/Web/Models/Entities/TicketComment.php
index 36d21d8e..c47a47e1 100644
--- a/Web/Models/Entities/TicketComment.php
+++ b/Web/Models/Entities/TicketComment.php
@@ -117,12 +117,16 @@ class TicketComment extends RowModel
     {
         $mark = $this->getMark();
 
-        if($mark === 0) {
-            return false;
-        } elseif ($mark === 1) {
-            return true;
-        } elseif ($mark === null) {
-            return null;
+        switch ($mark) {
+            case 0 :
+                return false;
+                break;
+            case 1 :
+                return true;
+                break;
+            default :
+                return null;
+                break;
         }
     }
 
diff --git a/Web/Models/Repositories/TicketComments.php b/Web/Models/Repositories/TicketComments.php
index 6201089c..94360aec 100644
--- a/Web/Models/Repositories/TicketComments.php
+++ b/Web/Models/Repositories/TicketComments.php
@@ -51,11 +51,11 @@ class TicketComments
 
     function get(int $id): ?TicketComment
     {
-        $comment = $this->comments->where(['id' => $id])->fetch();
+        $comment = $this->comments->get($id);;
         if (!is_null($comment))
             return new TicketComment($comment);
         else
-            return null;
+            return NULL;
     }
    
     use \Nette\SmartObject;