Athena connect to snowflake no driver issue

I followed the instructions from https://docs.aws.amazon.com/athena/latest/ug/connect-to-a-data-source-lambda.html to set up the connection between Athena and snowflake. However I'm getting an error when swtiching to the new data sources from Athena interface, the error msg is: Failed to invoke lambda function due to com.amazonaws.services.lambda.invoke.LambdaFunctionException: 0: java.sql.SQLException: No suitable driver found for jdbc:snowflake://<my connection string>

Am I missing something when deploying the connector? Thanks.