Small test error.

Removed unneeded code.