Added test for setting custom template base class