Fix relative imports

Signed-off-by: Harshavardhana <harsha@harshavardhana.net>