Fix test in production