Fix a type and add a unit tests heading