small safety fix