Updated test to accept API key.