Fix mismatch between spec and code