Fixed an issue I didn't notice before committing