fixed removal of code