fixed up minor issue after removing some comments