Fix build

Signed-off-by: Michael Bestas <mikeioannina@gmail.com>