Fixed small bug, thanks to unholysampler.