Fixed up tests to compile correctly.