Fixed another undefined method error