Fixes ws_entity
Changed entity to ws_entity