Whoops, fix this logic.