Update, seems to work!