Use tmp_dir for test