Updates so it actually works