Fix minor issue with variant