Fixed a couple of files