     /// - Parameters:
     ///   - avAudioUnit: AVAudioUnit to associate with
     ///   - index: Position of the parameter
-    public func associate(with avAudioUnit: AVAudioUnit, index: Int) {
-        self.avAudioUnit = avAudioUnit
-        parameter = avAudioUnit.auAudioUnit.parameterTree!.allParameters[index]
+    public func associate(with avAudioNode: AVAudioNode, index: Int) {
+        self.avAudioNode = avAudioNode
+        parameter = avAudioNode.auAudioUnit.parameterTree!.allParameters[index]
         assert(parameter != nil)
     }
 

/// NodeParameter wraps AUParameter in a user-friendly interface and adds some AudioKit-specific functionality.
/// New version for use with Parameter property wrapper.
public class NodeParameter {
    private var avAudioUnit: AVAudioUnit!

    /// AU Parameter that this wraps
    public private(set) var parameter: AUParameter!

    /// Stop automation
    public func stopAutomation() {
        if let token = renderObserverToken {
            avAudioNode.auAudioUnit.removeRenderObserver(token)
        }
    }

