fix: forgot class