Fix 504 error