[Issue #195][Fixed] added a fix to tests