Ehmm.. correct the mistake I did in the last commit (was actually harder to solve what I though)