diff --git a/ham.mac b/ham.mac
index dcbadb2..1b83730 100644
--- a/ham.mac
+++ b/ham.mac
@@ -9,7 +9,7 @@ ham(L, qs) := block([ps, qdots, H],
 	H: subst(solve(maplist(lambda([p,qdot], p = diff(L, qdot)), ps, qdots),
 	               qdots),
 	         apply("+", maplist("*", ps, qdots)) - L),
-	maplist(trigsimp, flatten(maplist(lambda([q], [
-		'diff(q,t) = diff(H, concat(p_, q)),
-		'diff(concat(p_, q),t) = -diff(H, q)
-	]), qs))));
+	maplist(trigsimp, maplist(lambda([q], [
+		diff(H, concat(p_, q)),
+        -diff(H, q)
+	]), qs)));