Fix a couple of type signatures.