Update code for fixed comment