diff --git a/README.md b/README.md
index ae79c83..600e8e3 100644
--- a/README.md
+++ b/README.md
@@ -1,7 +1,7 @@
 # Autonomous Racing Robot With an Arduino, a Raspberry Pi and a Pi Camera
 
-[![Build Status](https://travis-ci.com/sergionr2/RacingRobot.svg?branch=line-pred)](https://travis-ci.com/sergionr2/RacingRobot)
-[![codecov](https://codecov.io/gh/sergionr2/RacingRobot/branch/line-pred/graph/badge.svg)](https://codecov.io/gh/sergionr2/RacingRobot)
+[![Build Status](https://travis-ci.com/sergionr2/RacingRobot.svg?branch=master)](https://travis-ci.com/sergionr2/RacingRobot)
+[![codecov](https://codecov.io/gh/sergionr2/RacingRobot/branch/master/graph/badge.svg)](https://codecov.io/gh/sergionr2/RacingRobot)
 
 
 Autonomous toy racing car. CAMaleon team at the Toulouse Robot Race 2017. Humbavision team at IronCar.
diff --git a/train/utils.py b/train/utils.py
index ea6993d..231ff55 100644
--- a/train/utils.py
+++ b/train/utils.py
@@ -166,6 +166,7 @@ def __getitem__(self, index):
 
         # Normalize labels
         labels = np.array(self.labels[image]).astype(np.float32)
+        assert len(labels) == 3, "Error with label of image {}".format(image)
         labels[:, 0] = (labels[:, 0] - margin_left) / MAX_WIDTH
         labels[:, 1] = (labels[:, 1] - margin_top) / MAX_HEIGHT