One more tweak to add pro version support