Reword unit tests