Fixed a bit of my testing code.