Fix some stupid code