fixes #346 - Fixed a problem with cancelling units