Thanks @ElBrogrammer 

I didn't add this code, do you mind to create an issue to keep track on this? or just check if that actually works?