Fix code style in unit tests