Small bugfix, all good now