Fix compilation errors

Signed-off-by: Amitay Isaacs <amitay@gmail.com>