minor
Fixing tests