Set the right directory