diff --git a/pyformat.py b/pyformat.py
index 8f9f48c..de5e33b 100755
--- a/pyformat.py
+++ b/pyformat.py
@@ -231,10 +231,10 @@ def _main(argv, standard_out, standard_error):
                   file=standard_error)
             return 2
 
-    changed_and_error =  format_multiple_files(set(args.files),
-                                               args,
-                                               standard_out,
-                                               standard_error)
+    changed_and_error = format_multiple_files(set(args.files),
+                                              args,
+                                              standard_out,
+                                              standard_error)
     return 1 if changed_and_error[1] else 0
 
 
diff --git a/test_pyformat.py b/test_pyformat.py
index 07cf546..dbe5fdb 100755
--- a/test_pyformat.py
+++ b/test_pyformat.py
@@ -74,7 +74,7 @@ def test_format_code_with_remove_all_unused_imports(self):
 
     def test_format_code_with_remove_unused_variables(self):
         self.assertEqual(
-            "def test():\n    return 42\n",
+            'def test():\n    return 42\n',
             pyformat.format_code(
                 'def test():\n    x = 4\n    return 42',
                 aggressive=True,