wip: fixing complex case

Signed-off-by: Alex Lourie <djay.il@gmail.com>