[Fix] Fix the previous fix