fixed broken build. (missing class)