Getting return value of compute() into variable and call setCallSite(prevCallSite) only once. Adding return for other code paths (for None)