Fixes previously introduced bug