fixed my stupid code