fix ws issue - using dto instead of model