Fixing missing if-block.