Fix cube rewriter test