Damn, forgot to update a class name