diff --git a/gprMax/input_cmds_singleuse.py b/gprMax/input_cmds_singleuse.py
index 5a388fe0..952e9167 100644
--- a/gprMax/input_cmds_singleuse.py
+++ b/gprMax/input_cmds_singleuse.py
@@ -248,7 +248,10 @@ def process_singlecmds(singlecmds, multicmds, G):
             w = Waveform()
             w.ID = waveformIDs[waveform]
             w.type = 'user'
-            w.uservalues = waveformvalues[:,waveform]
+            if len(waveformvalues.shape) == 1:
+                w.uservalues = waveformvalues[:]
+            else:
+                w.uservalues = waveformvalues[:,waveform]
         
             if G.messages:
                 print('User waveform {} created.'.format(w.ID))