Minor. Just small cleanup