BUG: Fix kde caching

See if this looks okay.