Small bit fix