diff --git a/tests/Environment.js b/tests/Environment.js
index f02f904..02ef262 100644
--- a/tests/Environment.js
+++ b/tests/Environment.js
@@ -48,9 +48,9 @@ InputRecorder.prototype.generate = function() {
 
     for (var i=0; i<this.inputs.length; i++) {
         var input = this.inputs[i];
-        var numberOfChannels = input.length;
+        var numberOfChannels = input.samples.length;
         var recordBuffer = buffers[i];
-        for (var j=0; j<this.inputs[i]; j++) {
+        for (var j=0; j<numberOfChannels; j++) {
             if (j >= recordBuffer.length) {
                 recordBuffer.push([]);
             }
@@ -59,4 +59,10 @@ InputRecorder.prototype.generate = function() {
     }
 };
 
+InputRecorder.prototype.reset = function() {
+    for (var i=0; i<this.inputs.length; i++) {
+        this.buffers[i] = [];
+    }
+};
+