Fixed some left over test code.