fix broken test suite

This fixes the type signatures on the test suite so that they once again pass.