Trying to fix works on live site