Fix small Bug