Fix so it actually works