Fix a  couple of files