Attempt to fix code display