small error / bug fix