Whoops, duplicate code issue