diff --git a/ballerina/modules/wssec/document.bal b/ballerina/modules/wssec/document.bal
index 5a284a6..f467a4a 100644
--- a/ballerina/modules/wssec/document.bal
+++ b/ballerina/modules/wssec/document.bal
@@ -16,6 +16,7 @@
import ballerina/jballerina.java;
class Document {
+
private handle nativeDocumentBuilder;
public function init(xml xmlPayload) returns Error? {
diff --git a/ballerina/modules/wssec/sec_header.bal b/ballerina/modules/wssec/sec_header.bal
index e3f0537..92f4deb 100644
--- a/ballerina/modules/wssec/sec_header.bal
+++ b/ballerina/modules/wssec/sec_header.bal
@@ -16,6 +16,7 @@
import ballerina/jballerina.java;
class WSSecurityHeader {
+
private handle nativeSecHeader;
function init(Document document) returns Error? {
diff --git a/ballerina/modules/wssec/signature.bal b/ballerina/modules/wssec/signature.bal
index 1023f5a..f9fc76d 100644
--- a/ballerina/modules/wssec/signature.bal
+++ b/ballerina/modules/wssec/signature.bal
@@ -17,6 +17,7 @@ import ballerina/crypto;
import ballerina/jballerina.java;
class Signature {
+
private handle nativeSignature;
function init() returns Error? {
diff --git a/ballerina/modules/wssec/tests/test_utils.bal b/ballerina/modules/wssec/tests/test_utils.bal
index 4881ff4..2e99abe 100644
--- a/ballerina/modules/wssec/tests/test_utils.bal
+++ b/ballerina/modules/wssec/tests/test_utils.bal
@@ -13,7 +13,6 @@
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
-
import ballerina/test;
const string USERNAME = "username";
@@ -38,6 +37,7 @@ function assertTimestampToken(string envelopeString) {
test:assertTrue(envelopeString.includesMatch(created));
test:assertTrue(envelopeString.includesMatch(expires));
}
+
function assertUsernameToken(string envelopeString, PasswordType passwordType) {
string:RegExp usernameTokenTag = re `.*`;
string:RegExp usernameTag = re `${USERNAME}`;
@@ -64,6 +64,7 @@ function assertUsernameToken(string envelopeString, PasswordType passwordType) {
}
}
}
+
function assertSignatureWithX509(string securedEnvelope) {
string:RegExp keyIdentifier = re `.*`;
test:assertTrue(securedEnvelope.includesMatch(keyIdentifier));