Improve overriding base task classes