minor bugfix - forgot something