If we use it only in tests now we can move it to the test dir.