Completing the code that was added in the previous commit.