Improving unit tests for base entities.