nit use test.assertEqual here too