Fixed the NoMethodError