Fix step assignment with term sharing and partially applied function

Fix step assignment with term sharing and partially applied function #4522