diff --git a/migrations/v9.go b/migrations/v9.go
index 5ef991f..3766cbe 100644
--- a/migrations/v9.go
+++ b/migrations/v9.go
@@ -1,29 +1,33 @@
 /*
  * Copyright © 2020 A Bunch Tell LLC.
  *
  * This file is part of WriteFreely.
  *
  * WriteFreely is free software: you can redistribute it and/or modify
  * it under the terms of the GNU Affero General Public License, included
  * in the LICENSE file in this source code package.
  */
 
 package migrations
 
 func optimizeDrafts(db *datastore) error {
 	t, err := db.Begin()
+	if err != nil {
+		t.Rollback()
+		return err
+	}
 
 	_, err = t.Exec(`ALTER TABLE posts ADD INDEX(owner_id, id)`)
 	if err != nil {
 		t.Rollback()
 		return err
 	}
 
 	err = t.Commit()
 	if err != nil {
 		t.Rollback()
 		return err
 	}
 
 	return nil
 }