Should fix it

- Should now work as intended