Kleine bug fix