Fixed a couple of notices

Fixed a couple of notices