Fix minor bug in snake