tiny code fix