Fix bug test-only method