fix for bad removal of code