BUG: fixed bug due to a forgotten case