Updated tests for store