Small bugfix, regarding previous update