Allow custom token type