Add version constraint to fix build