Fixed bug in gameplay