Fixed an undefined method error in previous commit.