updated to work with tez-0.5.0-SNAPSHOT