Yet another small fix