fix simple bug