Fixed minor boo boo