Fix minor problem