Update tests for configuration class change