diff --git a/docs/src/plots.md b/docs/src/plots.md
index 6cf9c10..6ba5afe 100644
--- a/docs/src/plots.md
+++ b/docs/src/plots.md
@@ -3,6 +3,7 @@
 
 ## 1D Plotting Functions
 ```@docs
+ScatteringPlotsExt.plotOriginalSignal1D
 ScatteringPlotsExt.plotZerothLayer1D
 ScatteringPlotsExt.plotFirstLayer1DSingleWavelet
 ScatteringPlotsExt.gifFirstLayer1D
diff --git a/src/scatteringplots.jl b/src/scatteringplots.jl
index 0e9c278..f563c29 100644
--- a/src/scatteringplots.jl
+++ b/src/scatteringplots.jl
@@ -1,6 +1,6 @@
 """
-    plotZerothLayer1D(sf; saveTo=nothing, index=1)
-Function that plots the zeroth layer of the scattering transform at a specified example index. 
+    plotOriginalSignal1D(f; title="Original Signal", saveTo=nothing, index=1)
+Function that plots the original signal at a specified example index. 
 """
 function plotOriginalSignal1D(f; title="Original Signal", saveTo=nothing, index=1)
     plt = plot(f[:,1,index], title=title, legend=false, xlim=(0, length(f[:, 1, index])+1), color=:blue, margin=5Plots.mm, size=(720,480))
diff --git a/test/fluxtests.jl b/test/fluxtests.jl
index e986d77..b66777b 100644
--- a/test/fluxtests.jl
+++ b/test/fluxtests.jl
@@ -99,6 +99,9 @@
         @test res1[1:32*3, 1] ≈ reshape(res[0][:, :, 1], (32 * 3,))
     end
 
+    #=
+    As of this moment these tests are broken. This is merely because GPU support has not been added to
+    MonogenicFilterFlux.jl yet. When support is added these tests will be re-enabled. 
     nFilters = [1, 12, 12, 12]
     @testset "2D basics" begin
         n_init_channels=2
@@ -141,6 +144,7 @@
         @test size(res1) == (totalSize, 2)
         @test res1[1:32^2*n_init_channels, 1] ≈ reshape(res[0][:, :, :, 1], (32^2 * n_init_channels,))
     end
+    =#
 
     nFilters = [1, 10, 9]
     @testset "1D integer pooling" begin