While writing the test, we noticed it didn't work because the labels weren't correctly generated, so we're fixing them as well.