Correct last pull

Need more corrections because now the mod doesn't works but don't know which
Merge pull request #1 from Zaraki98/patch-1

Correct last pull