Fix reference problem