fixed bug - bad reference