Fixed tests due to change of deps