Fixed small issues introduced because I'm writing code on my break at 7/11 :guitar:.