Small error in tests fixed.