bug fix
modified protection