Fixed stupid code