Update configurations for the test
Remove unnecessary codes and fix retrieving incorrect key-name