Fix LWS issue

Refs #6