Revert "removed some test code"

This reverts commit 551e6e8c84f6996039ef870dd6d964fb272aa711.