minor type fix for unit tests