Fixed directory problem

fixes #6