Small bugfix for v 1.01