now, as it works correct, fix the test