Add version constraint to fix build error