Revert "remove test code"

This reverts commit d0f6c53947a982d0eac24325facd95567e2a0b8d.