Fixes/tweaks previous change

- It broke regular widgets, woops.