Made unit tests more specific