Added condition to check if stage is in url