adding support to set the type that you wanna use :thumbsup: