Fix unit tests to account for dot