register fix :)