FIX: minor issue in tests