small changes after tests