fix broken tests

cc @jonahkagan