Fixed some stray code