Made unit tests more clean.