Removed leftover test code

Apologies. Forgot to take this out from the previous pull request.