Small bugfix related to last commit.