NoMethodException problem is fixed