fix #4 wrong dataset directory