fixed the f'n bug