Updated dependency + Fixed code