Fix some issues with the live code editor