fixed bug - works now