Fixed stupid programming mistake (forgot class method parameters in def)