Fix last commit so it actually works