Fixed key error. Was returning the wrong key for the type of a comment