Fix test for TLS