Small change of if else block.