diff --git a/Commands/save shared should or context.tmCommand b/Commands/save shared should or context.tmCommand
index 49d74c2..57dafe6 100644
--- a/Commands/save shared should or context.tmCommand	
+++ b/Commands/save shared should or context.tmCommand	
@@ -10,7 +10,7 @@ input = STDIN.read
 
 if input =~ /shared_should/i or input =~ /shared_context_should/
 	out = input.sub('shared_should', '').sub('shared_context_should', '')
-	out = out.sub(/\s*do\s*$/, '')
+	out = out.sub(/\s*do\s*(|.*|)?\s*$/, '')
 	out = out.gsub(/[^\w_\s]/, '')
 	out = out.squeeze('_ 	').strip
 	out = out.gsub(/\s/, '_')