fixed broken class

my bad