Small bug fix (again :panda_face: )