diff --git a/scripts/train.sh b/scripts/train.sh
index d53b119..0ee2565 100644
--- a/scripts/train.sh
+++ b/scripts/train.sh
@@ -37,7 +37,7 @@ if [ ! -f "${DEST}" ]; then
 	wget -q -O "${DEST}" "${DOWNLOAD}"
 fi
 
-unzip "${DEST}" -d "${UNZIPPED}"
+unzip -o "${DEST}" -d "${UNZIPPED}"
 
 if [ ! -f "${UNZIPPED}/data.yaml" ]; then
 	fail "${ID}" "Invalid dataset"
@@ -56,7 +56,7 @@ WEIGHTS="yolov5s.pt"
 
 if [[ "${MODEL}" == "yolov5m_onnx" ]]; then
 	WEIGHTS="yolov5m.pt"
-else if [[ "${MODEL}" == "yolov5l_onnx" ]]; then
+elif [[ "${MODEL}" == "yolov5l_onnx" ]]; then
 	WEIGHTS="yolov5l.pt"
 fi