fixed theta update