Fixed checkstyle in unit tests