Fix type and add proper dependency