Add ability to use custom config for tests