Fixes headers

Signed-off-by: TD-D <orenfza@gmail.com>