Small buggfix