Small little fix