change test file to load right train data