Fix node map test

Cache SSA transformed asts at codegen