couple of little fixes