Fix an issue introduced in 189536e

Fix an issue introduced in 189536e