This should have been in last commit
Fixed broken class