Small update to remove logging