Keep the type when using a custom type