Minor bugfix with last commit.