Updated  to work with 1.4.1

Signed-off-by: Julio Capote <julio@twitter.com>