Changing params back to class instead of defined type to resolve inheritance issue