Dev/davide barbieri/bug fix