From 45dc9e5c0d1640b7e0c90da4f7aebd01561e21ee Mon Sep 17 00:00:00 2001
From: SebastianM-C <31181429+SebastianM-C@users.noreply.github.com>
Date: Mon, 7 May 2018 22:24:07 +0300
Subject: [PATCH] Add 2D marker_z with Functions

---
 src/pipeline.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/pipeline.jl b/src/pipeline.jl
index bca77623..110a29bc 100644
--- a/src/pipeline.jl
+++ b/src/pipeline.jl
@@ -117,7 +117,7 @@ function _preprocess_userrecipe(kw::KW)
     # map marker_z if it's a Function
     if isa(get(kw, :marker_z, nothing), Function)
         # TODO: should this take y and/or z as arguments?
-        kw[:marker_z] = map(kw[:marker_z], kw[:x], kw[:y], kw[:z])
+        kw[:marker_z] = isa(kw[:z], Void) ? map(kw[:marker_z], kw[:x], kw[:y]) : map(kw[:marker_z], kw[:x], kw[:y], kw[:z])
     end
 
     # map line_z if it's a Function