Mini-fix of dfs script