couple little fixes