Fixed a wee bug