Fix file not found error when running tests