Fix YAGNI