WCM-213 Fix bug