diff --git a/deep_speech_2/tests/test_setup.py b/deep_speech_2/tests/test_setup.py
index 71a46afb73455e4206c215ef2fd72c95b232b5d2..18b9c1a0ce5333f559383b18704edf7270457fcf 100644
--- a/deep_speech_2/tests/test_setup.py
+++ b/deep_speech_2/tests/test_setup.py
@@ -14,8 +14,8 @@ class TestSetup(unittest.TestCase):
         file = 'test.wav'
         sf.write(file, data, 44100, format='WAV', subtype='FLOAT')
         read, fs = sf.read(file)
-        assert np.all(read == data)
-        assert fs == 44100
+        self.assertTrue(np.all(read == data))
+        self.assertEqual(fs, 44100)
         os.remove(file)