Reverted the test code for #122.