refactor after color refinement test