Fix small fail

I forgot to include some files and select the right directory.