diff --git a/src/LDAWrap.cc b/src/LDAWrap.cc
index 34d4e17..c155bd2 100644
--- a/src/LDAWrap.cc
+++ b/src/LDAWrap.cc
@@ -17,8 +17,8 @@ void LDAWrap::Init(Local<Object> target) {
   ctor->InstanceTemplate()->SetInternalFieldCount(1);
   ctor->SetClassName(Nan::New("LDA").ToLocalChecked());
 
-  Nan::SetPrototypeMethod(ctor, "subspaceProject", SubspaceProject);
-  Nan::SetPrototypeMethod(ctor, "subspaceReconstruct", SubspaceReconstruct);
+  Nan::SetMethod(ctor, "subspaceProject", SubspaceProject);
+  Nan::SetMethod(ctor, "subspaceReconstruct", SubspaceReconstruct);
 
   target->Set(Nan::New("LDA").ToLocalChecked(), ctor->GetFunction());
 };
diff --git a/test/.unit.js.swp b/test/.unit.js.swp
deleted file mode 100644
index b3cd16c..0000000
Binary files a/test/.unit.js.swp and /dev/null differ
diff --git a/test/unit.js b/test/unit.js
index 0190704..b271ec4 100755
--- a/test/unit.js
+++ b/test/unit.js
@@ -314,6 +314,18 @@ test("fonts", function(t) {
   });
 })
 
+test('LDA Wrap', function(assert) {
+  // subspaceProject
+  var mat = cv.LDA.subspaceProject(new cv.Matrix(1, 2, cv.Constants.CV_64F), new cv.Matrix(), new cv.Matrix(2, 1, cv.Constants.CV_8UC1));
+  assert.deepEqual(mat.size(), [2,2], 'subspaceProject');
+
+  // subspaceReconstruct
+  mat = cv.LDA.subspaceReconstruct(new cv.Matrix(1, 2, cv.Constants.CV_64F), new cv.Matrix(), new cv.Matrix(1, 2, cv.Constants.CV_8UC1));
+  assert.deepEqual(mat.size(), [1,1], 'subspaceReconstruct');
+
+  assert.end();
+})
+
 // Test the examples folder.
 require('./examples')()