Fix data loading in visualisation tests