Fixed small bug introduced in last commit.