[BUGFIX] Some bug fix on the tests package