Update cache setter to fix tests