Minor bug-fix