Updated types in include file so dialyzer and tests pass.