Fix the undefined method error (obsolete)