fix for diskstore
modify unit test case