Temporary fix for code2notebook; need to fix or remove