diff --git a/thermostat.py b/thermostat.py
index 459ca04..cc5c879 100644
--- a/thermostat.py
+++ b/thermostat.py
@@ -201,13 +201,16 @@ class Set_mode_thermostat(Resource):
             return "OK", 201
         except sqlalchemy.exc.IntegrityError as e:
             try:
+              db.session.rollback()
               cur_mode = db.session.execute(db.select(Set_mode).filter_by(name="mode")).scalar_one()
               cur_mode = data
               db.session.commit()
             except Exception as e:
+              db.session.rollback()
               logging.error(e)
               return "K0", 400
         except Exception as e:
+            db.session.rollback()
             logging.error(e)
             return "K0", 400