Fix javascript error due to renamed class

@andrewgarner 

I need to wait for Dough PR before merging this.