fix a problem where we removed some lines of code accidentally