Fix minor thing