Revert "removed a couple lines"

This reverts commit 82b6558d0b607f8acda4db7faebd2f800df94051.