Bug fix

My last commit had a game-breaking bug. Fixed it