Fix typo in last commit
Hm... more flexibility but I'm not sure about this one