Bug fix (undefined method)