small adjustment to in class