Def to class fix

This obviously should've been a Class. I shouldn't program when I only
have a few minutes in the morning.