Imrpvoed TaskGenerator Tests