fix: forgot to add the new class whoops!