Fixes undefined method