Just another small fix