These statements:

a = iam()()
b = a()

a("s1")
b("s2")

should be equivalent to:

iam()()("s1")
iam()()()("s2")

but you code produces:

iam()()()("s1")
iam("s2")