Small bugfixing.