diff --git a/yapftests/yapf_test.py b/yapftests/yapf_test.py
index 6425370..a545008 100644
--- a/yapftests/yapf_test.py
+++ b/yapftests/yapf_test.py
@@ -34,7 +34,8 @@
 class FormatCodeTest(unittest.TestCase):
 
   def _Check(self, unformatted_code, expected_formatted_code):
-    formatted_code = yapf_api.FormatCode(unformatted_code)
+    formatted_code = yapf_api.FormatCode(unformatted_code,
+                                         style_config='chromium')
     self.assertEqual(expected_formatted_code, formatted_code)
 
   def testSimple(self):
@@ -45,9 +46,11 @@
 
   def testNoEndingNewline(self):
     unformatted_code = textwrap.dedent(u"""\
-        if True: pass""")
+        if True:
+          pass""")
     expected_formatted_code = textwrap.dedent(u"""\
-        if True: pass
+        if True:
+          pass
         """)
     self._Check(unformatted_code, expected_formatted_code)