Tweak test data import