Cleanup
Fix bad ref