Small bugfix, thanks to Gait.