Tweak rego function