Fixes

Signed-off-by: Spyros Lefkaditis <slefkaditis@gmail.com>