Lite buggfix