Small bug-fix