expanding unit tests slightly