That looks better You should also return S.One, not the Python integer. Someone else following this issue will have to decide if this is the right thing to do.