fixed failing tests due to bug fix