Review comments

Jira Issue: WP-778