Fix tests to point to right directory