Fixed bug when I used assertEqual rather than assertAlmostEqual