diff --git a/README.md b/README.md
index cd96f5f..f2ff5ba 100644
--- a/README.md
+++ b/README.md
@@ -5,7 +5,7 @@
 Easy to use implementation of various multi-object tracking algorithms.
 
 [![DOI](https://zenodo.org/badge/148338463.svg)](https://zenodo.org/badge/latestdoi/148338463)   
-[![Upload motrackers to PyPI](https://github.com/adipandas/multi-object-tracker/actions/workflows/python-publish.yml/badge.svg)](https://github.com/adipandas/multi-object-tracker/actions/workflows/python-publish.yml)
+<!-- [![Upload motrackers to PyPI](https://github.com/adipandas/multi-object-tracker/actions/workflows/python-publish.yml/badge.svg)](https://github.com/adipandas/multi-object-tracker/actions/workflows/python-publish.yml) -->
 
 
 `YOLOv3 + CentroidTracker` |  `TF-MobileNetSSD + CentroidTracker`
diff --git a/motrackers/detectors/yolo.py b/motrackers/detectors/yolo.py
index d6fd24d..ee169ec 100644
--- a/motrackers/detectors/yolo.py
+++ b/motrackers/detectors/yolo.py
@@ -23,7 +23,7 @@ def __init__(self, weights_path, configfile_path, labels_path, confidence_thresh
         object_names = load_labelsjson(labels_path)
 
         layer_names = self.net.getLayerNames()
-        if cv2.__version__ == '4.6.0':
+        if cv.__version__ >= '4.6.0':
             self.layer_names = [layer_names[i - 1] for i in self.net.getUnconnectedOutLayers()]
         else:
             self.layer_names = [layer_names[i[0] - 1] for i in self.net.getUnconnectedOutLayers()]
diff --git a/motrackers/sort_tracker.py b/motrackers/sort_tracker.py
index baaf174..0a3519d 100644
--- a/motrackers/sort_tracker.py
+++ b/motrackers/sort_tracker.py
@@ -140,7 +140,7 @@ def update(self, bboxes, detection_scores, class_ids):
                     self._remove_track(track_id)
         else:
             matches, unmatched_detections, unmatched_tracks = assign_tracks2detection_iou(
-                bbox_tracks, bbox_detections, iou_threshold=0.3)
+                bbox_tracks, bbox_detections, iou_threshold=self.iou_threshold)
 
             for i in range(matches.shape[0]):
                 t, d = matches[i, :]
diff --git a/motrackers/tracker.py b/motrackers/tracker.py
index da92b19..cbbff58 100644
--- a/motrackers/tracker.py
+++ b/motrackers/tracker.py
@@ -83,9 +83,10 @@ def _get_tracks(tracks):
         """
 
         outputs = []
-        for trackid, track in tracks.items():
-            if not track.lost:
-                outputs.append(track.output())
+        for _, track in tracks.items():
+            # if not track.lost:
+            #     outputs.append(track.output())
+            outputs.append(track.output())
         return outputs
 
     @staticmethod
diff --git a/pyproject.toml b/pyproject.toml
index 155c620..6ce1ac0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "motrackers"
-version = "0.0.1"
+version = "0.0.2"
 description = "Multi-object trackers in Python"
 authors = [
     {name = "Aditya M. Deshpande", email = "adityadeshpande2010@gmail.com"}