Fixed build issue

Class code was out-of-sync