I just took the opportunity to uniformize code, I think this test fits better.