Thanks for pointing the mistake. Corrected the tests now.

Solution to that problem is correct. Sets weren't introduced yet.