Fix deprecation warning

Signed-off-by: Amit Gupta <agupta@pivotallabs.com>