Add tests for restricted set
Really add them