Fix type problem