Slight fixup