Add missing class from recent commit 69abade