fixing comments on last pr