Add bugfix for pro feature