Fix type mismatch in last commit