fixed bug (hopefully)