bugfixing. simple but cool