Fix derp in previous commit
Make the code prettier :D