bugfix
better lookup of test/train