Fix set tests