Solving a minor issue caused by the last commit.