Allow tests to work with test