updated to fix a couple of issues with generated ko script