fix a bug @schmodt spotted

// cc @schmodt @adgad