Minor bugfix for earlier commit