Quick small fix