If anybody is having trouble with this, I managed to get it to work by doing the following (I have a turn based game with a button that ends the turn):
