Removing if/else for dig.