PyLint was complaining a bit, so I fixed this