Cleanups to transform code.