Fix for undefined method