Fixed little mistake in the code.

Thanks to Martin Hauke.