Fixed a couple of small mistakes