Fixed something, now it actually works as expected