Remove YAGNI code