Fixed so it actually works