tests: fix tests for changes of data_store