Fixing stats tests