Hopefully this works

I really dislike working from forks but I need to update my source