Change type to abstract class