Fix data restore