Fixed last commit

Yep, forgot to test it :D

It should be okay now