Update config for unit tests