Add test to ensure cache works