Revert "Removed some test code"

This reverts commit 68ac7d3ef413e9072979d05b68d1910b02d6fea5.