fixes .. bug