fixed class name

left over from when I was testing separately