Updates tests to be runnable for both /src and /dist