Error fixed

Error fixed that steals the improvements no matter what game type are
you playing.