Good point. Done. I've also simplified those tests - one assertion is enough for each test case.