Change condition to check for current_user.