Fix coding mistake