bug fixing

:)