Fixes NoMethodError