diff --git a/download_only.py b/download_only.py
index 8a1353f..3b9408e 100644
--- a/download_only.py
+++ b/download_only.py
@@ -74,7 +74,10 @@ def db_set_col(db, video_id, column, value=True):
     Also updates the updated_at timestamp.
     """
     cursor = db.cursor()
-    query = f"UPDATE videos SET {column} = %s WHERE id = %s"
+    if column == "data_downloaded":
+        query = f"UPDATE videos SET {column} = %s, download_end=NOW() WHERE id = %s"
+    else:
+        query = f"UPDATE videos SET {column} = %s WHERE id = %s"
     cursor.execute(query, (value, video_id))
     db.commit()
     cursor.close()
@@ -88,7 +91,7 @@ def try_lock_video(db, video_id):
     cursor = db.cursor()
     query = """
         UPDATE videos
-        SET data_downloading = 1, updated_at = NOW()
+        SET data_downloading = 1, updated_at = NOW(), download_start = NOW()
         WHERE id = %s AND data_downloading = 0
     """
     cursor.execute(query, (video_id,))