diff --git a/run.py b/run.py
index f16b3fbac466fccdb26253e5a009f278441f1733..b3749f659cd0d289e4a341640ddf381ed80e6616 100755
--- a/run.py
+++ b/run.py
@@ -63,9 +63,7 @@ print(result)
 
 driver.find_element_by_class_name('dialog_button').click() # Unnecessary.
 
-if result == 'Done successfully!':
-    driver.close()
-else:
-    exit(3)
+driver.close()
+exit(0 if result == 'Done successfully!' else 3)