Fixed a couple of issues I found while running the script