diff --git a/derivative.py b/derivative.py
index e3c8722e543d56139e0fffb9ab6ac6a2f7b2f495..fca8071132516c3bd23c23dbd01191a9fa3d18c9 100644
--- a/derivative.py
+++ b/derivative.py
@@ -152,7 +152,7 @@ class DiffOperator(sp.Expr):
             args = normalizeProduct(mul)
             diffs = [a for a in args if isinstance(a, DiffOperator)]
             if len(diffs) == 0:
-                return mul
+                return mul * argument
             rest = [a for a in args if not isinstance(a, DiffOperator)]
             diffs.sort(key=defaultDiffSortKey)
             result = argument
@@ -166,7 +166,7 @@ class DiffOperator(sp.Expr):
         elif expr.func == sp.Add:
             return expr.func(*[handleMul(a) for a in expr.args])
         else:
-            return expr
+            return expr * argument
 
 # ----------------------------------------------------------------------------------------------------------------------