Squashed a couple of small bugs.

Thanks as always,

Lewis 👨‍💻