Fixed up some incorrect code