Fixed some leftover test code