Fix remove dir bug