Now works with 2.1.1

Signed-off-by: Max Salazar <m.s.lechuga@gmail.com>